hackerrank prime dates problem solution in python